Searched refs:input_padding_value (Results 1 – 2 of 2) sorted by relevance
44 const auto& input_padding_value = model->GetArray(op->inputs[4]); in Run() local47 !input_padding_value.buffer) { in Run()63 input_padding_value.GetBuffer<ArrayDataType::kUint8>().data; in Run()
1224 TF_ASSIGN_OR_RETURN(HloInstruction * input_padding_value, in RewriteDynamicSelectAndScatterSamePadding()1242 input_padding_value); in RewriteDynamicSelectAndScatterSamePadding()1259 hlo, input, input_padding_value, absl::MakeSpan(padding_before), &window, in RewriteDynamicSelectAndScatterSamePadding()