Searched refs:ScalarIndexedConstantArray (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 304 return Construct<ScalarIndexedConstantArray>(constant, indices, source_dim, in ComputeArrayForGather()
600 const Shape& shape, ScalarIndexedConstantArray* operand) { in FoldReshapeOfGather()
616 operand_without_degenerate_dims->as<ScalarIndexedConstantArray>())); in FoldReshapeOfGather()
635 const Shape& shape, ScalarIndexedConstantArray* scalar_indexed) { in FoldReshapeOfGatherNoDegenerateDims()
809 dynamic_cast<ScalarIndexedConstantArray*>(operand)) { in ComputeArrayForReshape()
842 ScalarIndexedConstantArray* lhs_scalar_indexed_const = in ComputeArrayForElementwiseBinaryOp()
843 dynamic_cast<ScalarIndexedConstantArray*>(lhs); in ComputeArrayForElementwiseBinaryOp()
844 ScalarIndexedConstantArray* rhs_scalar_indexed_const = in ComputeArrayForElementwiseBinaryOp()
845 dynamic_cast<ScalarIndexedConstantArray*>(rhs); in ComputeArrayForElementwiseBinaryOp()
859 ScalarIndexedConstantArray* scalar_indexed_const = in ComputeArrayForElementwiseBinaryOp()
[all …]
|
| D | indexed_array_analysis.h | 215 class ScalarIndexedConstantArray : public ScalarIndexedArray {
224 explicit ScalarIndexedConstantArray(Array* source, Array* indices, in ScalarIndexedConstantArray() function
270 const PrecisionConfig& precision_config, ScalarIndexedConstantArray* lhs,
276 ScalarIndexedConstantArray* rhs);
319 const Shape& shape, ScalarIndexedConstantArray* operand);
321 const Shape& shape, ScalarIndexedConstantArray* scalar_indexed);
340 return Construct<ScalarIndexedConstantArray>(source, indices, source_dim, in ConstructScalarIndexedArray()
|