Searched defs:shape_1 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | util.cc | 42 std::vector<int> shape_1; in IsBroadcastable() local
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | sequential.py | 545 def relax_input_shape(shape_1, shape_2): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 130 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
159 Shape shape_1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | operation_selector.cc | 96 const BHWC shape_1{input_shape.b, 36, tiles_x * tiles_y, output_shape.c}; in WinogradFromNode() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 782 const Shape& shape_1, in HasCompatibleElementTypes()
795 const Shape& shape_1 = instruction->operand(1)->shape(); in HandleRng() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2236 const int64_t shape_1[] = {1, 3}; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 1272 auto shape_1 = ops::Shape(s.WithOpName("s"), conv_1); in TEST_F() local
|