Home
last modified time | relevance | path

Searched defs: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
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc158 ShapeHandle vector_shape = c->Vector(last_dim); in __anon3e672dd80402() local
2412 ShapeHandle vector_shape = c->Vector(channel_dim); in __anon3e672dd83e02() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc280 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
326 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
Dtuple_test.cc170 auto vector_shape = builder.GetShape(vector_element).ConsumeValueOrDie(); in XLA_TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc816 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc879 Shape vector_shape = ShapeUtil::MakeShape(F32, {17}); in TEST_F() local