Searched refs:kShapeAttr (Results 1 – 3 of 3) sorted by relevance
94 OP_REQUIRES_OK(context, context->GetAttr(kShapeAttr, &shape_)); in ImmutableConstantOp()111 constexpr char const* ImmutableConstantOp::kShapeAttr; member in tensorflow::ImmutableConstantOp
37 static constexpr char const* kShapeAttr = "shape"; variable
44 const char* const kShapeAttr = "_shape"; variable88 .Attr(kShapeAttr, TensorShape(feed.shape())) in AddArgNodes()241 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kShapeAttr, &shape)); in CreateXlaArgs()