| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test_common.h | 48 std::initializer_list<int> shape_shape, in ReshapeOpModel() argument
53 this->BuildWithTensorShape(input_shape, shape_shape, shape_data); in ReshapeOpModel()
56 this->BuildWithConstantTensorShape(input_shape, shape_shape, in ReshapeOpModel()
95 std::initializer_list<int> shape_shape, in BuildWithTensorShape() argument
99 int shape_input_tensor = this->AddInput({TensorType_INT32, shape_shape}); in BuildWithTensorShape()
115 std::initializer_list<int> shape_shape, in BuildWithConstantTensorShape() argument
119 this->AddConstInput(TensorType_INT32, shape_data, shape_shape); in BuildWithConstantTensorShape()
|
| D | broadcast_to_test.cc | 33 std::initializer_list<int> shape_shape) { in BroadcastToOpModel() argument
35 shape_ = AddInput({GetTensorType<ShapeType>(), shape_shape}); in BroadcastToOpModel()
40 BuildInterpreter({input_shape, shape_shape}); in BroadcastToOpModel()
45 std::initializer_list<int> shape_shape, in BroadcastToOpModel() argument
49 AddConstInput(GetTensorType<ShapeType>(), shape_values, shape_shape); in BroadcastToOpModel()
54 BuildInterpreter({input_shape, shape_shape}); in BroadcastToOpModel()
|
| D | scatter_nd.cc | 52 const RuntimeShape& shape_shape, in CheckShapes() argument
56 (shape_shape.DimensionsCount() == 1)); in CheckShapes()
65 shape_shape.Dims(0) - ix); in CheckShapes()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | empty_op.cc | 43 const TensorShape shape_shape = ctx->InputShape("shape"); in Compile() local
45 ctx, TensorShapeUtils::IsVector(shape_shape), in Compile()
47 shape_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_concat_op_test.py | 37 shape_shape=None): argument
44 dtypes.int64, shape=shape_shape))
335 self._SparseTensor_UnknownShape(shape_shape=[3])
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.h | 252 ShapeHandle values_shape, ShapeHandle shape_shape);
|
| D | common_shape_fns.cc | 2228 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor() argument
2233 TF_RETURN_IF_ERROR(c->WithRank(shape_shape, 1, &unused_shape)); in ValidateSparseTensor()
2253 DimensionHandle shape_rank_dim = c->Dim(shape_shape, 0); in ValidateSparseTensor()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test.cc | 923 Output shape_shape = ops::Const(root.WithOpName("shape_shape"), {2}, {1}); in TEST() local
925 ops::RandomUniformInt(root.WithOpName("shape"), shape_shape, in TEST()
994 Output shape_shape = in TEST() local
997 ops::RandomUniformInt(root.WithOpName("test/shape_rng"), shape_shape, in TEST()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 3244 shape_shape = self._sparse_tensor.dense_shape.get_shape()
3245 rank = (indices_shape.dims[1] - 1).merge_with(shape_shape.dims[0] - 1)
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2720 TensorShape shape_shape({var_shape.dims()}); in CreatePerHostDummyArgs() local
2721 shape_shape.AsProto(tensorshape_proto.mutable_tensor_shape()); in CreatePerHostDummyArgs()
|