Searched defs:SetAttrShape (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | abstract_operation.h | 130 inline Status AbstractOperation::SetAttrShape(const char* attr_name, in SetAttrShape() function
|
| H A D | c_api_unified_experimental_graph.cc | 201 Status SetAttrShape(const char* attr_name, const int64_t* dims, in SetAttrShape() function in tensorflow::tracing::graph::GraphOperation
|
| H A D | gradients.cc | 350 Status SetAttrShape(AbstractOperation* op_, const char* attr_name, in SetAttrShape() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| H A D | tape_operation.cc | 87 Status TapeOperation::SetAttrShape(const char* attr_name, const int64_t* dims, in SetAttrShape() function in tensorflow::gradients::TapeOperation
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | eager_operation.cc | 75 Status EagerOperation::SetAttrShape(const char* attr_name, const int64_t* dims, in SetAttrShape() function in tensorflow::EagerOperation
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| H A D | c_api_unified_experimental_mlir.cc | 470 Status MlirAbstractOp::SetAttrShape(const char* attr_name, const int64_t* dims, in SetAttrShape() function in mlir::TF::__anon3eda563f0111::MlirAbstractOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| H A D | op_handler_selector_test.cc | 241 Status SetAttrShape(const char* attr_name, const int64_t* dims, in SetAttrShape() function in tfrt::tf::__anon6788e1330111::FakeOperation
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 1670 tensorflow::Status OperationInterface::SetAttrShape(const char* attr_name, in SetAttrShape() function in tfrt::tf::OperationInterface
|