Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddensify.h34 std::vector<int> vector_shape(dims_count); in Densify()
36 vector_shape[i] = output_shape.Dims(i); in Densify()
39 tflite::optimize::sparsity::FormatConverter<T> converter(vector_shape, in Densify()
/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.cc283 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
286 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
288 ShapeUtil::MakeTupleShape({vector_shape, array_shape}); in XLA_TEST_F()
329 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
332 ShapeUtil::MakeTupleShape({array_shape, vector_shape}); in XLA_TEST_F()
334 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.cc155 ShapeHandle vector_shape = c->Vector(last_dim); in __anonad10f3360302() local
156 c->set_output(1, vector_shape); in __anonad10f3360302()
157 c->set_output(2, vector_shape); in __anonad10f3360302()
158 c->set_output(3, vector_shape); in __anonad10f3360302()
159 c->set_output(4, vector_shape); in __anonad10f3360302()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1165 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape() local
1166 c->set_output(1, vector_shape); in FusedBatchNormShape()
1167 c->set_output(2, vector_shape); in FusedBatchNormShape()
1168 c->set_output(3, vector_shape); in FusedBatchNormShape()
1169 c->set_output(4, vector_shape); in FusedBatchNormShape()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc3174 std::vector<int> vector_shape(dims_count); in PrepareOpsToDelegate() local
3176 vector_shape[i] = output_tensor.dims->data[i]; in PrepareOpsToDelegate()
3184 vector_shape, *input_tensor.sparsity); in PrepareOpsToDelegate()
3196 vector_shape, *input_tensor.sparsity); in PrepareOpsToDelegate()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc1326 Shape vector_shape = ShapeUtil::MakeShape(F32, {17}); in TEST_F() local
1328 ShapeInference::InferSliceShape(vector_shape, {2}, {4}, {1}); in TEST_F()