Searched defs:shape_elements (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | while_test.cc | 353 std::vector<Shape> shape_elements = { in XLA_TEST_F() local
406 std::vector<Shape> shape_elements = { in XLA_TEST_F() local
463 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
511 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
557 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
615 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
692 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
755 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
820 std::vector<Shape> shape_elements = {ShapeUtil::MakeShape(S32, {}), in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | post_quantize.cc | 308 DenseIntElementsAttr shape_elements; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1731 auto shape_elements = operands[1].dyn_cast_or_null<DenseElementsAttr>(); in fold() local
|