Home
last modified time | relevance | path

Searched defs:input_ref (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_util.cc158 TensorReference input_ref(*input); in DeviceToDeviceCopy() local
208 TensorReference input_ref(*device_tensor); in CopyPluggableDeviceTensorToCPU() local
257 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToPluggableDevice() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util.cc236 TensorReference input_ref(*input); in DeviceToDeviceCopy() local
298 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU() local
344 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU() local
/aosp_15_r20/external/cronet/net/android/
H A Dkeystore.cc71 ScopedJavaLocalRef<jbyteArray> input_ref = ToJavaByteArray(env, input); in SignWithPrivateKey() local
97 ScopedJavaLocalRef<jbyteArray> input_ref = ToJavaByteArray(env, input); in EncryptWithPrivateKey() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dvalue_refinement_utils.cpp206 auto& input_ref = info[n->input()]; in handleCommonRefinentOperators() local
225 auto& input_ref = info[non_const_input]; in handleCommonRefinentOperators() local
/aosp_15_r20/external/rust/android-crates-io/crates/rusqlite/src/
Dsession.rs225 let input_ref = &input; in invert_strm() localVariable
538 let input_ref = &input; in add_stream() localVariable
625 let input_ref = &input; in apply_strm() localVariable
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DTranspose.cpp57 ValueRef input_ref, in add_transpose_view_node()
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_ref.cpp154 TensorRef<Tensor<float, 3>> input_ref(input); in test_ref_in_expr() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfractional_max_pool_op.cc331 const T& input_ref = tensor_in_mat.coeffRef(d, in_index); in Compute() local
H A Dmaxpooling_op.cc159 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolWithArgMaxHelper() local
655 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad() local
H A Dpooling_ops_3d.cc650 const T& input_ref = in_mat.coeffRef(d, in_index); in launch() local
/aosp_15_r20/external/vulkan-validation-layers/tests/
H A Dvklayertests_descriptor_renderpass_framebuffer.cpp2766 VkAttachmentReference input_ref = {0, VK_IMAGE_LAYOUT_GENERAL}; in TEST_F() local
/aosp_15_r20/external/deqp/external/openglcts/modules/gl/
H A Dgl4cShadingLanguage420PackTests.cpp15958 std::string input_ref; in prepareDrawShaderSource() local