Home
last modified time | relevance | path

Searched defs:param_index (Results 1 – 24 of 24) 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/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/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/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/core/kernels/
Dtraining_ops_gpu.cu.cc50 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyAdagradKernel() local
89 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyProximalAdagradKernel() local
132 const Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyFtrlKernel() local
236 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyKerasMomentumKernel() local
816 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyAdadeltaKernel() local
/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/libtextclassifier/native/actions/
Dactions-suggestions.cc99 const int param_index, const Variant& param_value, in SetVectorOrScalarAsModelInput()
753 const int param_index = entry->value(); in SetupModelInput() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc475 int64_t param_number, const std::vector<int64>& param_index) { in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h393 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.c115 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.cc733 Status UpdateOrTransformParamInput(int param_index, const string& op, in UpdateOrTransformParamInput()
/external/swiftshader/third_party/SPIRV-Tools/source/diff/
Ddiff.cpp1610 for (size_t param_index = 0; param_index < shared_param_count; in MatchFunctionParamIds() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/diff/
Ddiff.cpp1610 for (size_t param_index = 0; param_index < shared_param_count; in MatchFunctionParamIds() local
/external/selinux/libsepol/cil/src/
Dcil_resolve_ast.c3213 enum cil_sym_index param_index = CIL_SYM_UNKNOWN; in cil_resolve_name_call_args() local
/external/mesa3d/src/mesa/main/
Dvarray.c3265 unsigned param_index = 0; in _mesa_InternalBindVertexBuffers() local