Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_comparison.cc786 Status element_result = in NearHelper() local
789 if (!element_result.ok()) { in NearHelper()
790 element_result = InvalidArgument("Array at shape index %s, %s", in NearHelper()
792 element_result.error_message()); in NearHelper()
794 return_status = element_result; in NearHelper()
797 AppendStatus(return_status, element_result.error_message()); in NearHelper()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc672 xla::XlaOp list_result, element_result; in Compile() local
675 ExecuteTensorListPopBack(list, &list_result, &element_result, in Compile()
680 ctx->SetTensorListOutput(1, element_result); in Compile()
682 ctx->SetOutput(1, element_result); in Compile()
Dtensor_list_utils.h105 xla::XlaOp* element_result,
Dtensor_list_utils.cc415 xla::XlaOp* element_result, in ExecuteTensorListPopBack() argument
455 *element_result = xla::Tuple(b, element_result_parts); in ExecuteTensorListPopBack()
457 *element_result = element_result_parts[0]; in ExecuteTensorListPopBack()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc400 auto element_result = AttrElementT::get( in ConstFoldBinaryOpDenseDense() local
403 if (!element_result) return {}; in ConstFoldBinaryOpDenseDense()
405 return DenseElementsAttr::get(type, element_result); in ConstFoldBinaryOpDenseDense()