Searched refs:result_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/examples/micro_speech/ |
| D | recognize_commands_test.cc | 79 const int result_dims[] = {2, 1, 4}; in TF_LITE_MICRO_TEST() local
81 result_data, tflite::testing::IntArrayFromInts(result_dims), -128.0f, in TF_LITE_MICRO_TEST()
176 const int result_dims[] = {2, 1, 4}; in TF_LITE_MICRO_TEST() local
178 result_data, tflite::testing::IntArrayFromInts(result_dims), -128.0f, in TF_LITE_MICRO_TEST()
198 const int result_dims[] = {2, 1, 4}; in TF_LITE_MICRO_TEST() local
200 result_data, tflite::testing::IntArrayFromInts(result_dims), -128.0f, in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | resampler_ops.cc | 197 std::vector<int64> result_dims, in BoundSamples() argument
225 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in BoundSamples()
229 xla::Broadcast(xla::Zero(ctx->builder(), warp_type), result_dims); in BoundSamples()
580 std::vector<int64> result_dims(warp_dims.begin(), warp_dims.end() - 1); in Compile() local
581 result_dims.push_back(data_channels); in Compile()
586 xla::BroadcastInDim(is_in_bound, result_dims, broadcasted_dims); in Compile()
590 xla::Broadcast(xla::Zero(ctx->builder(), data_type), result_dims); in Compile()
666 std::vector<int64> result_dims(warp_dims.begin(), warp_dims.end() - 1); in Compile() local
667 result_dims.push_back(2); in Compile()
671 BoundSamples(ctx, warp, warp_type, warp_shape, result_dims, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1070 DimensionVector result_dims; in CollapseFirstNDims() local
1071 result_dims.push_back(prefix_dim); in CollapseFirstNDims()
1073 std::back_inserter(result_dims)); in CollapseFirstNDims()
1075 result_dims); in CollapseFirstNDims()
|