Searched refs:unknown_index (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_op.h | 51 int unknown_index = -1; in Compute() local
56 ValidateSizes<int32>(sizes, &product, &unknown_index, in Compute()
61 ValidateSizes<int64>(sizes, &product, &unknown_index, in Compute()
70 if (unknown_index != -1) { in Compute()
93 shape.set_dim(unknown_index, missing); in Compute()
111 Status ValidateSizes(const Tensor& sizes, int64* product, int* unknown_index, in ValidateSizes() argument
114 *unknown_index = -1; in ValidateSizes()
121 if (*unknown_index != -1) { in ValidateSizes()
123 "Only one input size may be -1, not both ", *unknown_index, in ValidateSizes()
126 *unknown_index = d; in ValidateSizes()
|
| D | reshape_util.cc | 108 int unknown_index = -1; in ReshapeSparseTensor() local
114 context, unknown_index == -1, in ReshapeSparseTensor()
117 unknown_index, " and ", d)); in ReshapeSparseTensor()
118 unknown_index = d; in ReshapeSparseTensor()
128 if (unknown_index != -1) { in ReshapeSparseTensor()
142 output_shape.set_dim(unknown_index, missing); in ReshapeSparseTensor()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 55 int unknown_index = -1; in Compile() local
61 ctx, unknown_index == -1, in Compile()
63 unknown_index, " and ", d)); in Compile()
64 unknown_index = d; in Compile()
80 if (unknown_index != -1) { in Compile()
103 shape.set_dim(unknown_index, missing); in Compile()
112 << shape.DebugString() << ", unknown_index=" << unknown_index; in Compile()
127 if (unknown_index == -1) { in Compile()
157 if (dim == unknown_index) { in Compile()
168 dims_are_dynamic[unknown_index] = input_is_dynamic; in Compile()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_reshape_op.cc | 82 int unknown_index = -1; in Compute() local
87 ValidateSizes<int32>(sizes, &product, &unknown_index, in Compute()
92 ValidateSizes<int64>(sizes, &product, &unknown_index, in Compute()
101 if (unknown_index != -1) { in Compute()
124 shape.set_dim(unknown_index, missing); in Compute()
201 Status ValidateSizes(const Tensor& sizes, int64* product, int* unknown_index, in ValidateSizes() argument
204 *unknown_index = -1; in ValidateSizes()
211 if (*unknown_index != -1) { in ValidateSizes()
213 "Only one input size may be -1, not both ", *unknown_index, in ValidateSizes()
216 *unknown_index = d; in ValidateSizes()
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | input_data.py | 279 unknown_index = {'validation': [], 'testing': [], 'training': []}
297 unknown_index[set_index].append({'label': word, 'file': wav_path})
317 random.shuffle(unknown_index[set_index])
319 self.data_index[set_index].extend(unknown_index[set_index][:unknown_size])
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1376 int unknown_index = -1; in GetReshapeOutputType() local
1384 if (unknown_index != -1) in GetReshapeOutputType()
1391 unknown_index, dim.index())); in GetReshapeOutputType()
1393 unknown_index = dim.index(); in GetReshapeOutputType()
1413 if (unknown_index != -1) { in GetReshapeOutputType()
1434 output_ty_shape[unknown_index] = missing_dim; in GetReshapeOutputType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 604 int unknown_index = -1; in GetReshapeOutputType() local
612 if (unknown_index != -1) in GetReshapeOutputType()
616 unknown_index, dim.index())); in GetReshapeOutputType()
618 unknown_index = dim.index(); in GetReshapeOutputType()
638 if (unknown_index != -1) { in GetReshapeOutputType()
659 output_ty_shape[unknown_index] = missing_dim; in GetReshapeOutputType()
|