Home
last modified time | relevance | path

Searched refs:vector_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dindex_util_test.cc37 Shape vector_shape = ShapeUtil::MakeShape(F32, {100}); in TEST() local
39 IndexUtil::MultidimensionalIndexToLinearIndex(vector_shape, {42})); in TEST()
41 IndexUtil::LinearIndexToMultidimensionalIndex(vector_shape, 42); in TEST()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc280 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
283 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
285 ShapeUtil::MakeTupleShape({vector_shape, array_shape}); in XLA_TEST_F()
326 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
329 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
331 ShapeUtil::MakeTupleShape({inner_tuple_shape, vector_shape}); in XLA_TEST_F()
Dtuple_test.cc170 auto vector_shape = builder.GetShape(vector_element).ConsumeValueOrDie(); in XLA_TEST_F() local
179 ASSERT_TRUE(ShapeUtil::Equal(vector_shape, ShapeUtil::MakeShape(F32, {3}))); in XLA_TEST_F()
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc158 ShapeHandle vector_shape = c->Vector(last_dim); in __anon3e672dd80402() local
159 c->set_output(1, vector_shape); in __anon3e672dd80402()
160 c->set_output(2, vector_shape); in __anon3e672dd80402()
161 c->set_output(3, vector_shape); in __anon3e672dd80402()
162 c->set_output(4, vector_shape); in __anon3e672dd80402()
2412 ShapeHandle vector_shape = c->Vector(channel_dim); in __anon3e672dd83e02() local
2413 c->set_output(1, vector_shape); in __anon3e672dd83e02()
2414 c->set_output(2, vector_shape); in __anon3e672dd83e02()
2415 c->set_output(3, vector_shape); in __anon3e672dd83e02()
2416 c->set_output(4, vector_shape); in __anon3e672dd83e02()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc816 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape() local
817 c->set_output(1, vector_shape); in FusedBatchNormShape()
818 c->set_output(2, vector_shape); in FusedBatchNormShape()
819 c->set_output(3, vector_shape); in FusedBatchNormShape()
820 c->set_output(4, vector_shape); in FusedBatchNormShape()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc879 Shape vector_shape = ShapeUtil::MakeShape(F32, {17}); in TEST_F() local
881 ShapeInference::InferSliceShape(vector_shape, {2}, {4}, {1}); in TEST_F()