Home
last modified time | relevance | path

Searched defs:padding_arg_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/protobuf/tpu/
Ddynamic_padding.proto18 int32 padding_arg_index = 3; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_padding_mapper.cc126 const int32_t padding_arg_index = padding_proto.padding_arg_index(); in GetRemappedPaddings() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc353 int padding_arg_index = in CompileTFFunctionToHlo() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1645 int32_t padding_arg_index, in AddDynamicParameterBindingEntry()
1726 const int32_t padding_arg_index = padding_arg_index_attr.getInt(); in AddDynamicParameterBindings() local