Searched refs:make_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_test.cc | 1610 auto make_shape = [&c](std::initializer_list<int64> dim_sizes) { in TestMergeHandles() local
1635 std::vector<ShapeAndType> t{{make_shape({1, 2, 3}), DT_FLOAT}, in TestMergeHandles()
1637 {make_shape({4, 3, 2, 1}), DT_INT32}}; in TestMergeHandles()
1650 0, std::vector<ShapeAndType>{{make_shape({1, 2, 3}), DT_FLOAT}})); in TestMergeHandles()
1664 t2[2].shape = make_shape({4, 3, 4, 1}); in TestMergeHandles()
1686 t[1].shape = make_shape({1, 10}); in TestMergeHandles()
1721 auto make_shape = [&c](std::initializer_list<int64> dim_sizes) { in TestRelaxHandles() local
1746 std::vector<ShapeAndType> t{{make_shape({1, 2, 3}), DT_FLOAT}, in TestRelaxHandles()
1748 {make_shape({4, 3, 2, 1}), DT_INT32}}; in TestRelaxHandles()
1761 0, std::vector<ShapeAndType>{{make_shape({1, 2, 3}), DT_FLOAT}})); in TestRelaxHandles()
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | execute.py | 135 def make_shape(v, arg_name): function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 3375 auto make_shape = [](absl::Span<const int64> dims, in TEST_F() local
3383 auto in_shape = make_shape(in_dims, options.input_minor_to_major_layout); in TEST_F()
3384 auto f_shape = make_shape(f_dims, options.filter_minor_to_major_layout); in TEST_F()
3385 auto out_shape = make_shape(out_dims, options.output_minor_to_major_layout); in TEST_F()
|