Searched refs:scalar_indexed_const (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 859 ScalarIndexedConstantArray* scalar_indexed_const = in ComputeArrayForElementwiseBinaryOp() local
882 if (!absl::c_all_of(scalar_indexed_const->output_dims(), in ComputeArrayForElementwiseBinaryOp()
899 absl::Span<const int64> output_dims = scalar_indexed_const->output_dims(); in ComputeArrayForElementwiseBinaryOp()
905 InsertAt(&simulated_index, scalar_indexed_const->source_dim(), in ComputeArrayForElementwiseBinaryOp()
923 scalar_indexed_const->source()->shape(), new_inner_broadcast_dims)); in ComputeArrayForElementwiseBinaryOp()
931 opcode, scalar_indexed_const->literal(), inner_broadcast_result))); in ComputeArrayForElementwiseBinaryOp()
936 opcode, inner_broadcast_result, scalar_indexed_const->literal()))); in ComputeArrayForElementwiseBinaryOp()
941 new_source, scalar_indexed_const->indices(), in ComputeArrayForElementwiseBinaryOp()
942 scalar_indexed_const->source_dim(), in ComputeArrayForElementwiseBinaryOp()
943 std::vector<int64>(scalar_indexed_const->output_dims().begin(), in ComputeArrayForElementwiseBinaryOp()
[all …]
|