Searched refs:shape_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test.cc | 50 std::initializer_list<int> shape_shape, in ReshapeOpModel() argument
55 BuildWithTensorShape(input_shape, shape_shape, shape_data); in ReshapeOpModel()
58 BuildWithConstantTensorShape(input_shape, shape_shape, shape_data); in ReshapeOpModel()
92 std::initializer_list<int> shape_shape, in BuildWithTensorShape() argument
96 int shape_input_tensor = AddInput({TensorType_INT32, shape_shape}); in BuildWithTensorShape()
109 std::initializer_list<int> shape_shape, in BuildWithConstantTensorShape() argument
113 AddConstInput(TensorType_INT32, shape_data, shape_shape); in BuildWithConstantTensorShape()
|
| /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 | 318 ShapeHandle values_shape, ShapeHandle shape_shape);
|
| D | common_shape_fns.cc | 1495 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor() argument
1500 TF_RETURN_IF_ERROR(c->WithRank(shape_shape, 1, &unused_shape)); in ValidateSparseTensor()
1520 DimensionHandle shape_rank_dim = c->Dim(shape_shape, 0); in ValidateSparseTensor()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test.cc | 791 Output shape_shape = ops::Const(root.WithOpName("shape_shape"), {2}, {1}); in TEST() local
793 ops::RandomUniformInt(root.WithOpName("shape"), shape_shape, in TEST()
862 Output shape_shape = in TEST() local
865 ops::RandomUniformInt(root.WithOpName("test/shape_rng"), shape_shape, in TEST()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 2132 shape_shape = self._sparse_tensor.dense_shape.get_shape()
2133 rank = (indices_shape.dims[1] - 1).merge_with(shape_shape.dims[0] - 1)
|