Home
last modified time | relevance | path

Searched defs:param_index (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config_test.cc38 const ShapeIndex& param_index, in expect_aliased()
55 const ShapeIndex& param_index, in expect_not_aliased()
Dhlo_input_output_alias_config.cc30 const ShapeIndex& param_index, in SetUpAlias()
86 ShapeIndex param_index(entry.parameter_shape_index().begin(), in CreateFromProto() local
Dhlo_input_output_alias_config.h80 const ShapeIndex& param_index) const { in ParameterHasAlias()
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_shader_llvm_tess.c176 LLVMValueRef param_index, ubyte name) in get_dw_address_from_generic_indices()
216 LLVMValueRef param_index) in get_tcs_tes_buffer_address()
247 LLVMValueRef param_index, in get_tcs_tes_buffer_address_from_generic_indices()
380 LLVMValueRef vertex_index, LLVMValueRef param_index, in si_nir_load_tcs_varyings()
428 LLVMValueRef vertex_index, LLVMValueRef param_index, in si_nir_load_input_tes()
464 LLVMValueRef vertex_index, LLVMValueRef param_index, in si_nir_store_output_tcs()
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_function.cpp105 size_t param_index = 0; in ValidateFunctionParameter() local
259 for (size_t argument_index = 3, param_index = 2; in ValidateFunctionCall() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/val/
Dvalidate_function.cpp105 size_t param_index = 0; in ValidateFunctionParameter() local
259 for (size_t argument_index = 3, param_index = 2; in ValidateFunctionCall() local
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_function.cpp105 size_t param_index = 0; in ValidateFunctionParameter() local
259 for (size_t argument_index = 3, param_index = 2; in ValidateFunctionCall() local
/external/tensorflow/tensorflow/python/framework/
Dpython_api_info.cc238 const int param_index = in InitializeAttribute() local
306 const ParamIndex param_index = in InitializeInput() local
Dpython_api_parameter_converter.cc158 const PythonAPIInfo& api_info, int param_index, in ConvertToTensorInPlace()
/external/tensorflow/tensorflow/python/ops/v1_compat_tests/
Dgradient_checker_test.py206 def _BuildAndTestMiniMNIST(self, param_index, tag): argument
/external/tensorflow/tensorflow/python/ops/
Dnn_batchnorm_test.py119 param_index, argument
161 param_index, argument
Dgradient_checker_v2_test.py283 def _BuildAndTestMiniMNIST(self, param_index, tag): argument
/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_gpu.cu.cc128 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyAdagradKernel() local
167 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyProximalAdagradKernel() local
210 const Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyFtrlKernel() local
312 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyKerasMomentumKernel() local
/external/libtextclassifier/native/actions/
Dactions-suggestions.cc97 const int param_index, const Variant& param_value, in SetVectorOrScalarAsModelInput()
751 const int param_index = entry->value(); in SetupModelInput() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc409 int64 param_number, const std::vector<int64>& param_index) { in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h416 ShapeIndex param_index; member
/external/mesa3d/src/amd/vulkan/
Dradv_nir_to_llvm.c430 LLVMValueRef param_index) in get_tcs_tes_buffer_address()
465 LLVMValueRef param_index; in get_tcs_tes_buffer_address_params() local
549 LLVMValueRef param_index, in store_tcs_output()
618 LLVMValueRef param_index, in load_tes_input()
/external/libaom/libaom/av1/encoder/
Dglobal_motion.c117 static int32_t add_param_offset(int param_index, int32_t param_value, in add_param_offset()
/external/mesa3d/src/compiler/glsl/
Dir.h955 int param_index; member
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer.cc732 Status UpdateOrTransformParamInput(int param_index, const string& op, in UpdateOrTransformParamInput()
/external/selinux/libsepol/cil/src/
Dcil_resolve_ast.c3132 enum cil_sym_index param_index = CIL_SYM_UNKNOWN; in cil_resolve_name_call_args() local
/external/llvm-project/lldb/source/Plugins/TypeSystem/Clang/
DTypeSystemClang.cpp7524 for (unsigned param_index = 0; param_index < num_params; ++param_index) { in AddMethodToCXXRecordType() local
7951 for (unsigned param_index = 0; param_index < num_args; ++param_index) { in AddMethodToObjCObjectType() local
/external/mesa3d/src/mesa/main/
Dvarray.c3265 unsigned param_index = 0; in _mesa_InternalBindVertexBuffers() local