Searched refs:MakeShapeWithType (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 117 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest()
118 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2}); in RunTest()
148 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest()
149 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest()
182 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest()
183 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2}); in RunTest()
218 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4}); in RunTest()
219 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 3, 3}); in RunTest()
276 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 5}); in RunTest()
277 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest()
[all …]
|
| D | dot_operation_test.cc | 192 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 4}), "arg0"); in XLA_TYPED_TEST()
194 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({4, 1}), "arg1"); in XLA_TYPED_TEST()
553 auto x = Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2, 2}), in XLA_TYPED_TEST()
555 auto y = Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2, 2}), in XLA_TYPED_TEST()
610 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "x"); in XLA_TYPED_TEST()
612 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "y"); in XLA_TYPED_TEST()
648 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "x"); in XLA_TYPED_TEST()
649 auto y = Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2}), "y"); in XLA_TYPED_TEST()
681 auto x = Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2}), "x"); in XLA_TYPED_TEST()
683 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "y"); in XLA_TYPED_TEST()
[all …]
|
| D | matrix_ops_simple_test.cc | 80 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({}), "x_value"); in XLA_TYPED_TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 392 static Shape MakeShapeWithType(absl::Span<const int64> dimensions) { in MakeShapeWithType() function
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 75 xla::Shape dummy = xla::ShapeUtil::MakeShapeWithType<float>({}); in ParseTupleNode()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 1368 Shape input_shape = ShapeUtil::MakeShapeWithType<float>(input_dims); in TEST_P()
1369 Shape filter_shape = ShapeUtil::MakeShapeWithType<float>(filter_dims); in TEST_P()
|