Searched refs:update_slice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | all_reduce_test.cc | 121 …%constant.9 = s64[] constant(0), metadata={op_type="xla::update_slice" source_file="__format__@ten… in XLA_TEST_F()
122 …t.8, u8[1]{0} %reshape.3, s64[] %constant.9), metadata={op_type="xla::update_slice" source_file="_… in XLA_TEST_F()
163 …%constant.9 = s64[] constant(0), metadata={op_type="xla::update_slice" source_file="__format__@ten… in XLA_TEST_F()
164 ….8, s32[1]{0} %reshape.3, s64[] %constant.9), metadata={op_type="xla::update_slice" source_file="_… in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 110 auto update_slice = element; in SetElement() local
112 update_slice = builder.create<TF::ReshapeOp>( in SetElement()
119 ArrayRef<Value>{buffer, update_slice, in SetElement()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 272 HloInstruction * update_slice, in ScatterLoopBody()
275 ElideDegenerateDims(update_slice, {0})); in ScatterLoopBody()
|