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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_hlo_support_checker_test.cc 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_hlo_support_checker_test.cc 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2);
  /external/tensorflow/tensorflow/compiler/xla/
layout_util_test.cc 37 Shape MakeShapeWithSparseLayout(PrimitiveType element_type,
96 Shape src = MakeShapeWithSparseLayout(F32, {2, 3}, 2);
139 {MakeShapeWithSparseLayout(F32, {2, 3}, 4),
140 MakeShapeWithSparseLayout(F32, {42, 123}, 4),
143 MakeShapeWithSparseLayout(F32, {1, 2, 3}, 6)})});
164 Shape src = MakeShapeWithSparseLayout(F32, {123, 42, 7}, 6);
181 Shape dst = MakeShapeWithSparseLayout(F32, {2, 3}, 4);
shape_util.h 329 static Shape MakeShapeWithSparseLayout(
shape_util.cc 205 /* static */ Shape ShapeUtil::MakeShapeWithSparseLayout(
578 result = ShapeUtil::MakeShapeWithSparseLayout(primitive_type, dimensions,
    [all...]
shape_util_test.cc 133 Shape expected = ShapeUtil::MakeShapeWithSparseLayout(F32, {123, 456}, 10);
literal_util.h     [all...]

Completed in 230 milliseconds