Home
last modified time | relevance | path

Searched refs:MakeShapeWithType (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc117 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()
313 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in RunTest()
314 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in RunTest()
[all …]
Dconvolution_test_1d.cc86 Shape input_shape = ShapeUtil::MakeShapeWithType<T>(input_dims); in TestImpl()
87 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>(filter_dims); in TestImpl()
240 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 5}); in RunTest()
241 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest()
341 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 5}); in RunTest()
342 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2}); in RunTest()
Ddot_operation_test.cc199 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 4}), "arg0"); in XLA_TYPED_TEST()
201 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({4, 1}), "arg1"); in XLA_TYPED_TEST()
597 auto x = Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2, 2}), in XLA_TYPED_TEST()
599 auto y = Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2, 2}), in XLA_TYPED_TEST()
654 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "x"); in XLA_TYPED_TEST()
656 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "y"); in XLA_TYPED_TEST()
690 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "x"); in XLA_TYPED_TEST()
691 auto y = Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2}), "y"); in XLA_TYPED_TEST()
722 auto x = Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2}), "x"); in XLA_TYPED_TEST()
724 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "y"); in XLA_TYPED_TEST()
[all …]
Dmatrix_ops_simple_test.cc80 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({}), "x_value"); in XLA_TYPED_TEST()
/external/tensorflow/tensorflow/compiler/xla/python/
Djax_jit.cc366 xla::Shape shape = xla::ShapeUtil::MakeShapeWithType<CppType>({}); in ConvertToScalarBuffer()
621 xla::Shape shape = xla::ShapeUtil::MakeShapeWithType<xla::complex128>({}); in HandleComplex()
630 xla::Shape shape = xla::ShapeUtil::MakeShapeWithType<xla::complex64>({}); in HandleComplex()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h454 static Shape MakeShapeWithType(absl::Span<const int64> dimensions) { in MakeShapeWithType() function
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h79 xla::Shape dummy = xla::ShapeUtil::MakeShapeWithType<float>({}); in ParseTupleNode()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1445 Shape input_shape = ShapeUtil::MakeShapeWithType<float>(input_dims); in TEST_P()
1446 Shape filter_shape = ShapeUtil::MakeShapeWithType<float>(filter_dims); in TEST_P()