Searched refs:OpMetadata (Results 1 – 25 of 25) sorted by relevance
29 Status InvalidParameterArgument(const OpMetadata& op_metadata, in InvalidParameterArgument()55 const OpMetadata& metadata = param->metadata(); in InvalidParameterArgument()
79 absl::optional<const OpMetadata*> ParameterMetadata( in ParameterMetadata()139 absl::optional<const OpMetadata*> metadata = in CompileExecutable()146 const OpMetadata& m = *metadata.value(); in CompileExecutable()
83 const OpMetadata& metadata);
125 OpMetadata metadata; in TEST()
1345 void set_metadata(const OpMetadata& metadata) { metadata_ = metadata; } in set_metadata()1346 const OpMetadata& metadata() const { return metadata_; } in metadata()1792 OpMetadata metadata_;1812 string OpMetadataToString(const OpMetadata& metadata);
556 OpMetadata b_metadata; in TEST_F()562 OpMetadata c_metadata; in TEST_F()
231 const OpMetadata& metadata) { in MakeCountedLoop()
267 bool ParseMetadata(OpMetadata* metadata);677 optional<OpMetadata> metadata; in ParseInstructionRhs()2854 OpMetadata result; in ParseAttributeHelper()2858 static_cast<optional<OpMetadata>*>(attr_out_ptr)->emplace(result); in ParseAttributeHelper()3719 bool HloParser::ParseMetadata(OpMetadata* metadata) { in ParseMetadata()
57 xla.OpMetadata metadata = 7;
691 OpMetadata metadata; in TEST_F()
3024 string OpMetadataToString(const OpMetadata& metadata) { in OpMetadataToString()
254 %typemap(in) const OpMetadata& (OpMetadata temp) {255 StatusOr<OpMetadata> statusor = numpy::OpMetadataFromPyObject($input);
83 StatusOr<OpMetadata> OpMetadataFromPyObject(PyObject* o);
349 StatusOr<OpMetadata> OpMetadataFromPyObject(PyObject* o) { in OpMetadataFromPyObject()350 OpMetadata result; in OpMetadataFromPyObject()
210 void SetOpMetadata(const OpMetadata& metadata);
72 OpMetadata = collections.namedtuple('OpMetadata', _OP_METADATA_FIELDS) variable238 return OpMetadata(
416 void ComputationBuilder::SetOpMetadata(const OpMetadata& metadata) { in SetOpMetadata()
38 OpMetadata metadata_;
95 xla::OpMetadata metadata; in Compute()
173 xla::OpMetadata retval_metadata; in BuildComputation()783 xla::OpMetadata arg_metadata; in BuildArguments()
54 const OpMetadata& metadata) { in CreateCholesky()
44 const OpMetadata& metadata) { in CreateCudnnConv()
119 OpMetadata metadata; in TEST_F()
153 void SetOpMetadata(const OpMetadata& metadata) { metadata_ = metadata; } in SetOpMetadata()699 OpMetadata metadata_;
243 message OpMetadata { message