Home
last modified time | relevance | path

Searched refs:DimensionOrConstant (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h39 struct DimensionOrConstant;
70 friend struct DimensionOrConstant;
124 struct DimensionOrConstant { struct
127 DimensionOrConstant(DimensionHandle dim);
130 DimensionOrConstant(int64 val);
137 DimensionOrConstant(); argument
364 static inline int64 Value(DimensionOrConstant d) { in Value()
367 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown()
459 ShapeHandle MakeShape(std::initializer_list<DimensionOrConstant> dims);
471 ShapeHandle Vector(DimensionOrConstant dim);
[all …]
Dcommon_shape_fns.h155 DimensionOrConstant filter_size,
165 DimensionOrConstant filter_size, int64 dilation_rate, int64 stride,
221 Status MakeShapeFromFormat(TensorFormat format, DimensionOrConstant N,
222 const std::vector<DimensionOrConstant>& spatial,
223 DimensionOrConstant C, ShapeHandle* out,
Dshape_inference.cc737 std::initializer_list<DimensionOrConstant> dims) { in MakeShape()
740 for (const DimensionOrConstant& d : dims) { in MakeShape()
766 ShapeHandle InferenceContext::Vector(DimensionOrConstant dim) { in Vector()
770 ShapeHandle InferenceContext::Matrix(DimensionOrConstant dim1, in Matrix()
771 DimensionOrConstant dim2) { in Matrix()
1013 DimensionOrConstant divisor, in Divide()
1037 Status InferenceContext::Add(DimensionHandle first, DimensionOrConstant second, in Add()
1064 DimensionOrConstant second, in Subtract()
1087 DimensionOrConstant second, in Multiply()
1115 Status InferenceContext::Min(DimensionHandle first, DimensionOrConstant second, in Min()
[all …]
Dcommon_shape_fns.cc141 shape_inference::DimensionOrConstant filter_size, int64 dilation_rate, in GetWindowedOutputSizeFromDimsV2()
188 shape_inference::DimensionOrConstant filter_size, int64 stride, in GetWindowedOutputSizeFromDims()
328 Status MakeShapeFromFormat(TensorFormat format, DimensionOrConstant N, in MakeShapeFromFormat()
329 const std::vector<DimensionOrConstant>& spatial, in MakeShapeFromFormat()
330 DimensionOrConstant C, ShapeHandle* out, in MakeShapeFromFormat()
Dshape_inference_test.cc115 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
/external/tensorflow/tensorflow/core/ops/
Dbatch_ops.cc69 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anon130e45de0102()
70 shape_inference::DimensionOrConstant(3)})})); in __anon130e45de0102()
Darray_ops.cc296 passed_shape, 0, c->MakeDim(shape_inference::DimensionOrConstant(1)), in __anon7c94107b0202()