Searched refs:CreateFromDimensions (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 120 LiteralUtil::CreateFromDimensions(index_shape.element_type(), {0}))); in ExpandIndexVectorIntoOperandSpace()
125 LiteralUtil::CreateFromDimensions(index_shape.element_type(), {1}))); in ExpandIndexVectorIntoOperandSpace()
|
| D | scatter_expander.cc | 141 LiteralUtil::CreateFromDimensions(index_shape.element_type(), {0}))); in ExpandIndexVectorIntoOperandSpace()
146 LiteralUtil::CreateFromDimensions(index_shape.element_type(), {1}))); in ExpandIndexVectorIntoOperandSpace()
|
| D | hlo_constant_folding_test.cc | 128 auto literal = LiteralUtil::CreateFromDimensions(F32, dimensions); in TEST_F()
|
| D | hlo_evaluator.cc | 473 auto result_literal = LiteralUtil::CreateFromDimensions( in HandleConcatenate()
|
| D | hlo_parser.cc | 2176 *literal = LiteralUtil::CreateFromDimensions( in ParseDenseLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | exhaustive_op_test.cc | 259 Literal input_literal = LiteralUtil::CreateFromDimensions(ty, {input_size}); in RunImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 85 /* static */ Literal LiteralUtil::CreateFromDimensions( in CreateFromDimensions() function in xla::LiteralUtil
|
| D | literal_util.h | 251 static Literal CreateFromDimensions(PrimitiveType primitive_type,
|
| D | literal_test.cc | 174 LiteralUtil::CreateFromDimensions(S32, {2, 2, 1, 1, 1, 2}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | numpy_bridge.cc | 417 auto literal = LiteralUtil::CreateFromDimensions( in XlaLiteralFromPyObject()
|