Searched refs:InputRange (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 139 Status OpKernel::InputRange(StringPiece input_name, int* start, in InputRange() function in tensorflow::OpKernel
353 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in input()
371 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in input_dtype()
389 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in input_ref_mutex()
473 params_->op_kernel->InputRange(input_name, &input_index, &stop)); in forward_input_to_output_with_shape()
596 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in mutable_input()
621 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in replace_ref_input()
637 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in input_list()
645 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in mutable_input_list()
|
| D | op_kernel_test.cc | 947 TF_CHECK_OK(op->InputRange(input_name, &start, &stop)); in BM_InputRangeHelper()
|
| D | op_kernel.h | 172 Status InputRange(StringPiece input_name, int* start, int* stop) const;
|
| /external/deqp/external/vulkancts/modules/vulkan/spirv_assembly/ |
| D | vktSpvAsmTypeTests.cpp | 80 enum InputRange enum
708 InputRange inputRange,
716 InputRange inputRange,
724 InputRange inputRange,
732 InputRange inputRange,
776 InputRange inputRange);
781 InputRange inputRange);
786 InputRange inputRange);
791 InputRange inputRange);
923 InputRange inputRange) in combine()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.cc | 122 TF_RETURN_IF_ERROR(context->op_kernel().InputRange(name, &start, &stop)); in InputIndex()
360 TF_RETURN_IF_ERROR(op_kernel().InputRange(name, &start, &stop)); in ConstantInputList()
|
| D | xla_op_registry.cc | 414 TF_CHECK_OK(op_kernel->InputRange(input, &start, &stop)); in CompileTimeConstantInputs()
|