MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Scatter

Function Scatter

tensorflow/compiler/xla/client/xla_builder.cc:3561–3568  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3559}
3560
3561XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices,
3562 const XlaOp updates, const XlaComputation& update_computation,
3563 const ScatterDimensionNumbers& dimension_numbers,
3564 bool indices_are_sorted, bool unique_indices) {
3565 return input.builder()->Scatter(input, scatter_indices, updates,
3566 update_computation, dimension_numbers,
3567 indices_are_sorted, unique_indices);
3568}
3569
3570void Send(const XlaOp operand, const ChannelHandle& handle) {
3571 return operand.builder()->Send(operand, handle);

Callers 5

ExportXlaOpFunction · 0.50
XlaScatterFunction · 0.50
ScatterToGradDataFunction · 0.50
TEST_FFunction · 0.50
TEST_FFunction · 0.50

Calls 2

ScatterMethod · 0.45
builderMethod · 0.45

Tested by 2

TEST_FFunction · 0.40
TEST_FFunction · 0.40