Home
last modified time | relevance | path

Searched refs:target_param_index (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Ddynamic_parameter_binding.mlir37 // TUPLE-NEXT: target_param_index: 0
64 // TUPLE-NEXT: target_param_index: 0
68 // TUPLE-NEXT: target_param_index: 0
102 // TUPLE-NEXT: target_param_index: 0
106 // TUPLE-NEXT: target_param_index: 0
111 // TUPLE-NEXT: target_param_index: 3
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_parameter_binding.cc72 ShapeIndex target_param_index(binding.target_param_index().begin(), in CreateFromProto() local
73 binding.target_param_index().end()); in CreateFromProto()
78 DynamicDimension{target_param_num, target_param_index, in CreateFromProto()
Dhlo.proto340 // the subshape `target_param_index` of parameter `target_param_num`
360 // target_param_index = {}
366 repeated int64 target_param_index = 4; field
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc379 ShapeIndex target_param_index, in SetDynamicBinding() argument
389 for (int64 index : target_param_index) { in SetDynamicBinding()
410 target_param_num, target_param_index, target_dim_num})); in SetDynamicBinding()
Dxla_builder.h391 ShapeIndex target_param_index, int64 target_dim_num);