Searched refs:unknown_index (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_op.h | 47 int unknown_index = -1; in Compute() local
51 &unknown_index, &shape)); in Compute()
55 &unknown_index, &shape)); in Compute()
63 if (unknown_index != -1) { in Compute()
76 shape.set_dim(unknown_index, missing); in Compute()
94 Status ValidateSizes(const Tensor& sizes, int64* product, int* unknown_index, in ValidateSizes() argument
97 *unknown_index = -1; in ValidateSizes()
103 if (*unknown_index != -1) { in ValidateSizes()
105 "Only one input size may be -1, not both ", *unknown_index, in ValidateSizes()
108 *unknown_index = d; in ValidateSizes()
|
| D | mkl_reshape_op.cc | 84 int unknown_index = -1; in Compute() local
88 &unknown_index, &shape)); in Compute()
92 &unknown_index, &shape)); in Compute()
100 if (unknown_index != -1) { in Compute()
113 shape.set_dim(unknown_index, missing); in Compute()
222 Status ValidateSizes(const Tensor& sizes, int64* product, int* unknown_index, in ValidateSizes() argument
225 *unknown_index = -1; in ValidateSizes()
231 if (*unknown_index != -1) { in ValidateSizes()
233 "Only one input size may be -1, not both ", *unknown_index, in ValidateSizes()
236 *unknown_index = d; in ValidateSizes()
|
| D | reshape_util.cc | 60 int unknown_index = -1; in Reshape() local
66 context, unknown_index == -1, in Reshape()
69 unknown_index, " and ", d)); in Reshape()
70 unknown_index = d; in Reshape()
80 if (unknown_index != -1) { in Reshape()
93 output_shape.set_dim(unknown_index, missing); in Reshape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reshape_op.cc | 53 int unknown_index = -1; in Compile() local
58 ctx, unknown_index == -1, in Compile()
60 unknown_index, " and ", d)); in Compile()
61 unknown_index = d; in Compile()
71 if (unknown_index != -1) { in Compile()
84 shape.set_dim(unknown_index, missing); in Compile()
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | input_data.py | 276 unknown_index = {'validation': [], 'testing': [], 'training': []}
294 unknown_index[set_index].append({'label': word, 'file': wav_path})
314 random.shuffle(unknown_index[set_index])
316 self.data_index[set_index].extend(unknown_index[set_index][:unknown_size])
|