Searched refs:CreateFromDimensions (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 95 LiteralUtil::CreateFromDimensions(F32, outfeed_shape.dimensions())); in TEST_F()
107 &xla_builder_, LiteralUtil::CreateFromDimensions(F32, {2, 4, 5})); in TEST_F()
109 &xla_builder_, LiteralUtil::CreateFromDimensions(F32, {2, 6, 5})); in TEST_F()
120 LiteralUtil::CreateFromDimensions(F32, {3, 7})); in TEST_F()
122 LiteralUtil::CreateFromDimensions(F32, {})); in TEST_F()
133 LiteralUtil::CreateFromDimensions(F32, {3, 7})); in TEST_F()
135 LiteralUtil::CreateFromDimensions(F32, {})); in TEST_F()
147 LiteralUtil::CreateFromDimensions(F32, {3, 7})); in TEST_F()
158 LiteralUtil::CreateFromDimensions(F32, {3, 7})); in TEST_F()
|
| /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 | 130 auto literal = LiteralUtil::CreateFromDimensions(F32, dimensions); in TEST_F()
|
| D | while_loop_all_reduce_code_motion.cc | 372 HloInstruction::CreateConstant(LiteralUtil::CreateFromDimensions( in CreateNewWhileInit()
|
| D | hlo_evaluator.cc | 510 auto result_literal = LiteralUtil::CreateFromDimensions( in HandleConcatenate()
|
| D | hlo_parser.cc | 3122 *literal = LiteralUtil::CreateFromDimensions( in ParseDenseLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 213 static Literal CreateFromDimensions(PrimitiveType primitive_type,
|
| D | literal_util.cc | 85 /* static */ Literal LiteralUtil::CreateFromDimensions( in CreateFromDimensions() function in xla::LiteralUtil
|
| D | literal_test.cc | 184 LiteralUtil::CreateFromDimensions(S32, {2, 2, 1, 1, 1, 2}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | exhaustive_op_test_utils.h | 450 literals[i] = LiteralUtil::CreateFromDimensions(T, {GetInputSize()});
|