HomeSort by relevance Sort by last modified time
    Searched refs:MakeShapeWithType (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
convolution_test.cc 117 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2});
118 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 1, 2});
148 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4});
149 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2});
182 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4});
183 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 2, 2});
218 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 4, 4});
219 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 1, 3, 3});
276 Shape input_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 5});
277 Shape filter_shape = ShapeUtil::MakeShapeWithType<T>({1, 2, 2})
    [all...]
dot_operation_test.cc 192 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 4}), "arg0");
194 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({4, 1}), "arg1");
553 auto x = Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2, 2}),
555 auto y = Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2, 2}),
610 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "x");
612 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "y");
648 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "x");
649 auto y = Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2}), "y");
681 auto x = Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({2, 2}), "x");
683 Parameter(&builder, 1, ShapeUtil::MakeShapeWithType<T>({2, 2, 2}), "y")
    [all...]
matrix_ops_simple_test.cc 80 Parameter(&builder, 0, ShapeUtil::MakeShapeWithType<T>({}), "x_value");
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 392 static Shape MakeShapeWithType(absl::Span<const int64> dimensions) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_state_ops.h 75 xla::Shape dummy = xla::ShapeUtil::MakeShapeWithType<float>({});
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator_test.cc     [all...]

Completed in 557 milliseconds