| /external/protobuf/src/google/protobuf/compiler/cpp/ |
| D | metadata_test.cc | 78 std::string output_base = TestTempDir() + "/" + StripProto(filename); in CaptureMetadata() local
82 File::GetContents(output_base + ".pb.cc", pb_cc, true)); in CaptureMetadata()
87 File::GetContents(output_base + ".pb.h", pb_h, true)); in CaptureMetadata()
88 if (!atu::DecodeMetadata(output_base + ".pb.h.meta", pb_h_info)) { in CaptureMetadata()
94 GOOGLE_CHECK_OK(File::GetContents(output_base + ".proto.h", proto_h, in CaptureMetadata()
96 if (!atu::DecodeMetadata(output_base + ".proto.h.meta", proto_h_info)) { in CaptureMetadata()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 218 T* output_base = &output_mapper(0, col); in operator() local
219 typename TTypes<T>::UnalignedTensor output(output_base, num_rows); in operator()
245 T* output_base = &output_mapper(0, col);
246 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
284 T* output_base = &output_mapper(0, col);
285 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
327 T* output_base = &output_mapper(0, col);
328 typename TTypes<T>::UnalignedTensor output(output_base, num_rows);
|
| D | depthwise_conv_op.h | 148 const int64 output_base = i * padded_filter_stride;
152 Eigen::internal::pstoreu<T>(padded_filter + output_base + j, v);
156 padded_filter[output_base + vectorized_size + j] =
161 padded_filter[output_base + vectorized_size + scalar_size + j] =
|
| D | ragged_tensor_to_tensor_op.cc | 451 VALUE_TYPE* output_base = output_tensor->flat<VALUE_TYPE>().data(); in SetOutput() local
507 VALUE_TYPE* dst = output_base + dst_start * value_element_size; in SetOutput()
520 std::fill(output_base + dst_end * value_element_size, in SetOutput()
521 output_base + dst_i * value_element_size, *default_value); in SetOutput()
525 VALUE_TYPE* dst = output_base + dst_end * value_element_size; in SetOutput()
|
| D | depthwise_conv_grad_op.cc | 951 const int64 output_base = i * out_depth; in operator ()() local
955 auto out_block_data = filter_backprop + output_base + j; in operator ()()
974 filter_backprop[output_base + vectorized_size + j] += in operator ()()
|
| D | eigen_contraction_kernel.h | 709 ResScalar* output_base = &output_mapper(0, col); \
712 OutputRowMap(output_base, rows).setZero(); \
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_query.c | 188 nir_ssa_def *output_base = nir_imul(&b, output_stride, global_id); in build_occlusion_query_shader() local
252 store->src[2] = nir_src_for_ssa(output_base); in build_occlusion_query_shader()
263 store->src[2] = nir_src_for_ssa(output_base); in build_occlusion_query_shader()
273 nir_iadd(&b, result_size, output_base), in build_occlusion_query_shader()
352 nir_ssa_def *output_base = nir_imul(&b, output_stride, global_id); in build_pipeline_statistics_query_shader() local
372 nir_iadd(&b, output_base, nir_imul(&b, elem_count, elem_size)), in build_pipeline_statistics_query_shader()
377 nir_store_var(&b, output_offset, output_base, 0x1); in build_pipeline_statistics_query_shader()
449 nir_ssa_def *output_elem = nir_iadd(&b, output_base, in build_pipeline_statistics_query_shader()
559 nir_ssa_def *output_base = nir_imul(&b, output_stride, global_id); in build_tfb_query_shader() local
638 store->src[2] = nir_src_for_ssa(output_base); in build_tfb_query_shader()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 55 se::DeviceMemoryBase output_base = in ExecuteOnStream() local
65 config_, operand, output_base, scale, offset, mean, variance, &stream)); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/tools/ci_build/builds/ |
| D | docker_cpu_pip.sh | 43 bazel --output_base=/tmp test --define=no_tensorflow_py_deps=true \
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 456 VALUE_TYPE* output_base = GetTensorData<VALUE_TYPE>(output_tensor); in SetOutputT() local
493 VALUE_TYPE* dst = output_base + dst_start * value_element_size; in SetOutputT()
506 std::fill(output_base + dst_end * value_element_size, in SetOutputT()
507 output_base + dst_i * value_element_size, *default_value); in SetOutputT()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 851 xla::Literal output_base = xla::LiteralUtil::CreateR1<int32>({7, 42}); in TEST_F() local
855 xla::LiteralUtil::MakeTuple({&output_base, &output_grad1, &output_grad2}); in TEST_F()
|