Home
last modified time | relevance | path

Searched refs:SizeOfDimension (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/lite/kernels/
Dlsh_projection.cc82 TF_LITE_ENSURE(context, SizeOfDimension(hash, 1) <= 32); in Resize()
90 TF_LITE_ENSURE_EQ(context, SizeOfDimension(weight, 0), in Resize()
91 SizeOfDimension(input, 0)); in Resize()
98 outputSize->data[0] = SizeOfDimension(hash, 0); in Resize()
101 outputSize->data[0] = SizeOfDimension(hash, 0) * SizeOfDimension(hash, 1); in Resize()
117 int input_item_bytes = input->bytes / SizeOfDimension(input, 0); in RunningSignBit()
124 for (int i = 0; i < SizeOfDimension(input, 0); ++i) { in RunningSignBit()
144 int num_hash = SizeOfDimension(hash, 0); in SparseLshProjection()
145 int num_bits = SizeOfDimension(hash, 1); in SparseLshProjection()
159 int num_hash = SizeOfDimension(hash, 0); in DenseLshProjection()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dlsh_projection.cc82 TF_LITE_ENSURE(context, SizeOfDimension(hash, 1) <= 32); in Resize()
90 TF_LITE_ENSURE_EQ(context, SizeOfDimension(weight, 0), in Resize()
91 SizeOfDimension(input, 0)); in Resize()
98 outputSize->data[0] = SizeOfDimension(hash, 0); in Resize()
101 outputSize->data[0] = SizeOfDimension(hash, 0) * SizeOfDimension(hash, 1); in Resize()
117 int input_item_bytes = input->bytes / SizeOfDimension(input, 0); in RunningSignBit()
124 for (int i = 0; i < SizeOfDimension(input, 0); ++i) { in RunningSignBit()
144 int num_hash = SizeOfDimension(hash, 0); in SparseLshProjection()
145 int num_bits = SizeOfDimension(hash, 1); in SparseLshProjection()
159 int num_hash = SizeOfDimension(hash, 0); in DenseLshProjection()
[all …]
Dembedding_lookup.cc64 outputSize->data[0] = SizeOfDimension(lookup, 0); in Prepare()
65 outputSize->data[1] = SizeOfDimension(value, 1); in Prepare()
67 outputSize->data[i] = SizeOfDimension(value, i); in Prepare()
75 const int row_size = SizeOfDimension(value, 0); in EvalFloat()
78 for (int i = 0; i < SizeOfDimension(lookup, 0); i++) { in EvalFloat()
98 const int row_size = SizeOfDimension(value, 0); in EvalHybrid()
104 col_size *= SizeOfDimension(value, i); in EvalHybrid()
107 for (int i = 0; i < SizeOfDimension(lookup, 0); i++) { in EvalHybrid()
Dembedding_lookup_sparse.cc100 TF_LITE_ENSURE_EQ(context, SizeOfDimension(indices, 0), in Prepare()
101 SizeOfDimension(ids, 0)); in Prepare()
102 TF_LITE_ENSURE_EQ(context, SizeOfDimension(indices, 0), in Prepare()
103 SizeOfDimension(weights, 0)); in Prepare()
148 const int lookup_rank = SizeOfDimension(indices, 1); in Eval()
150 const int num_lookups = SizeOfDimension(ids, 0); in Eval()
151 const int num_rows = SizeOfDimension(value, 0); in Eval()
158 TF_LITE_ENSURE_EQ(context, SizeOfDimension(dense_shape, 0), lookup_rank); in Eval()
171 const int dim = SizeOfDimension(value, i); in Eval()
Dhashtable_lookup.cc72 TF_LITE_ENSURE_EQ(context, SizeOfDimension(key, 0), in Prepare()
73 SizeOfDimension(value, 0)); in Prepare()
81 hitSize->data[0] = SizeOfDimension(lookup, 0); in Prepare()
89 outputSize->data[0] = SizeOfDimension(lookup, 0); in Prepare()
91 outputSize->data[i] = SizeOfDimension(value, i); in Prepare()
108 const int num_rows = SizeOfDimension(value, 0); in Eval()
113 for (int i = 0; i < SizeOfDimension(lookup, 0); i++) { in Eval()
Dtranspose_conv.cc121 const int input_depth = SizeOfDimension(input, 3); in ResizeIm2ColTensor()
122 const int filter_width = SizeOfDimension(weights, 2); in ResizeIm2ColTensor()
123 const int filter_height = SizeOfDimension(weights, 1); in ResizeIm2ColTensor()
158 TF_LITE_ENSURE_EQ(context, SizeOfDimension(input, 3), in Prepare()
159 SizeOfDimension(weights, 3)); in Prepare()
198 const int width = SizeOfDimension(output, 2); in Eval()
199 const int height = SizeOfDimension(output, 1); in Eval()
200 const int filter_width = SizeOfDimension(weights, 2); in Eval()
201 const int filter_height = SizeOfDimension(weights, 1); in Eval()
Ddepthwise_conv.cc103 params->depth_multiplier * SizeOfDimension(input, 3), in Prepare()
104 SizeOfDimension(filter, 3)); in Prepare()
122 TF_LITE_ENSURE_EQ(context, SizeOfDimension(filter, 3), in Prepare()
123 SizeOfDimension(bias, 0)); in Prepare()
126 int channels_out = SizeOfDimension(filter, 3); in Prepare()
127 int width = SizeOfDimension(input, 2); in Prepare()
128 int height = SizeOfDimension(input, 1); in Prepare()
129 int filter_width = SizeOfDimension(filter, 2); in Prepare()
130 int filter_height = SizeOfDimension(filter, 1); in Prepare()
131 int batches = SizeOfDimension(input, 0); in Prepare()
Dreverse_sequence.cc80 TF_LITE_ENSURE_EQ(context, SizeOfDimension(seq_lengths_tensor, 0), in ReverseSequenceImpl()
81 SizeOfDimension(input, batch_dim)); in ReverseSequenceImpl()
83 TF_LITE_ENSURE(context, seq_lengths[i] <= SizeOfDimension(input, seq_dim)); in ReverseSequenceImpl()
Dsparse_to_dense.cc69 TF_LITE_ENSURE_EQ(context, SizeOfDimension(indices, 1), in CheckDimensionsMatch()
72 TF_LITE_ENSURE_EQ(context, SizeOfDimension(indices, 0), in CheckDimensionsMatch()
105 const int true_dimensions = SizeOfDimension(indices, 1); in GetIndicesVector()
205 const int num_indices = SizeOfDimension(indices, 0); in SparseToDenseImpl()
Dshape.cc32 output_data[i] = SizeOfDimension(input, i); in ExtractShape()
68 TFLITE_DCHECK_EQ(SizeOfDimension(output, 0), NumDimensions(input)); in Eval()
Dselect.cc55 SizeOfDimension(input_condition, 0) == SizeOfDimension(input_x, 0); in SelectPrepare()
Dkernel_util.h27 inline int SizeOfDimension(const TfLiteTensor* t, int dim) { in SizeOfDimension() function
53 count *= SizeOfDimension(t, i); in NumElements()
Dslice.cc57 size_value = SizeOfDimension(input, idx) - GetTensorData<T>(begin)[idx]; in CalculateOutputShapeVector()
59 if (SizeOfDimension(input, idx) < in CalculateOutputShapeVector()
Dmirror_pad.cc148 shape->data[i] = SizeOfDimension(input, i) + left_pad + right_pad; in GetPaddedOutputShape()
246 TF_LITE_ENSURE_EQ(context, SizeOfDimension(padding_matrix, 0), in Prepare()
Dkernel_util.cc221 int64_t d1 = i >= dims1 ? 1 : SizeOfDimension(input1, dims1 - i - 1); in CalculateShapeForBroadcast()
222 int64_t d2 = i >= dims2 ? 1 : SizeOfDimension(input2, dims2 - i - 1); in CalculateShapeForBroadcast()
Dpad.cc73 TF_LITE_ENSURE_EQ(context, SizeOfDimension(op_context->paddings, 0), in ResizeOutputTensor()
75 TF_LITE_ENSURE_EQ(context, SizeOfDimension(op_context->paddings, 1), 2); in ResizeOutputTensor()
Dgather_nd.cc64 const int indices_nd = SizeOfDimension(indices, indices_rank - 1); in Prepare()
Dsplit.cc56 const int input_size = SizeOfDimension(input, axis_value); in ResizeOutputTensors()
Dsvdf.cc167 TF_LITE_ENSURE_EQ(context, SizeOfDimension(activation_state, 0), batch_size); in Prepare()
168 TF_LITE_ENSURE_EQ(context, SizeOfDimension(activation_state, 1), in Prepare()
Dconv.cc254 TF_LITE_ENSURE_EQ(context, NumElements(bias), SizeOfDimension(filter, 0)); in Prepare()
597 const int input_size = NumElements(input) / SizeOfDimension(input, 0); in EvalHybrid()
598 const int batch_size = SizeOfDimension(input, 0); in EvalHybrid()
Dsplit_v.cc93 const int input_size = SizeOfDimension(input, axis_value); in ResizeOutputTensors()
Darg_min_max.cc44 output_dims->data[j] = SizeOfDimension(input, i); in ResizeOutput()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Ddepthwise_conv.cc167 int width = SizeOfDimension(input, 2); in Eval()
168 int height = SizeOfDimension(input, 1); in Eval()
169 int filter_width = SizeOfDimension(filter, 2); in Eval()
170 int filter_height = SizeOfDimension(filter, 1); in Eval()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder.cc68 const int batch_size = SizeOfDimension(inputs, 1); in Prepare()
169 const int max_time = SizeOfDimension(inputs, 0); in Eval()
170 const int batch_size = SizeOfDimension(inputs, 1); in Eval()
171 const int num_classes = SizeOfDimension(inputs, 2); in Eval()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc398 int width = SizeOfDimension(input, 2); in Eval()
399 int height = SizeOfDimension(input, 1); in Eval()
400 int filter_width = SizeOfDimension(filter, 2); in Eval()
401 int filter_height = SizeOfDimension(filter, 1); in Eval()

12