Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.h37 struct DimensionOrConstant;
68 friend struct DimensionOrConstant;
117 struct DimensionOrConstant { struct
120 DimensionOrConstant(DimensionHandle dim);
123 DimensionOrConstant(int64 val);
130 DimensionOrConstant(); argument
350 static inline int64 Value(DimensionOrConstant d) { in Value()
353 static inline bool ValueKnown(DimensionOrConstant d) { in ValueKnown()
445 ShapeHandle MakeShape(std::initializer_list<DimensionOrConstant> dims);
457 ShapeHandle Vector(DimensionOrConstant dim);
[all …]
Dcommon_shape_fns.h32 DimensionOrConstant filter_size,
42 DimensionOrConstant filter_size, int64 dilation_rate, int64 stride,
101 Status MakeShapeFromFormat(TensorFormat format, DimensionOrConstant N,
102 const std::vector<DimensionOrConstant>& spatial,
103 DimensionOrConstant C, ShapeHandle* out,
Dshape_inference.cc679 std::initializer_list<DimensionOrConstant> dims) { in MakeShape()
682 for (const DimensionOrConstant& d : dims) { in MakeShape()
708 ShapeHandle InferenceContext::Vector(DimensionOrConstant dim) { in Vector()
712 ShapeHandle InferenceContext::Matrix(DimensionOrConstant dim1, in Matrix()
713 DimensionOrConstant dim2) { in Matrix()
986 DimensionOrConstant divisor, in Divide()
1010 Status InferenceContext::Add(DimensionHandle first, DimensionOrConstant second, in Add()
1037 DimensionOrConstant second, in Subtract()
1060 DimensionOrConstant second, in Multiply()
1088 Status InferenceContext::Min(DimensionHandle first, DimensionOrConstant second, in Min()
[all …]
Dcommon_shape_fns.cc38 shape_inference::DimensionOrConstant filter_size, int64 dilation_rate, in GetWindowedOutputSizeFromDimsV2()
85 shape_inference::DimensionOrConstant filter_size, int64 stride, in GetWindowedOutputSizeFromDims()
494 Status MakeShapeFromFormat(TensorFormat format, DimensionOrConstant N, in MakeShapeFromFormat()
495 const std::vector<DimensionOrConstant>& spatial, in MakeShapeFromFormat()
496 DimensionOrConstant C, ShapeHandle* out, in MakeShapeFromFormat()
Dshape_inference_test.cc123 TEST_F(ShapeInferenceTest, DimensionOrConstant) { in TEST_F() argument
/external/tensorflow/tensorflow/core/ops/
Dbatch_ops.cc88 {c->MakeShape({shape_inference::DimensionOrConstant(c->UnknownDim()), in __anond951767c0102()
89 shape_inference::DimensionOrConstant(3)})})); in __anond951767c0102()
Dparsing_ops.cc49 shape_inference::DimensionOrConstant rank(c->UnknownDim()); in AddSparseOutputShapes()
Darray_ops.cc302 passed_shape, 0, c->MakeDim(shape_inference::DimensionOrConstant(1)), in __anon42d741190202()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc202 shape_inference::DimensionOrConstant in __anonbbc1c2d10202()
205 shape_inference::DimensionOrConstant in __anonbbc1c2d10202()
236 shape_inference::DimensionOrConstant lhs_batch_dimension_or_constant( in __anonbbc1c2d10202()
238 shape_inference::DimensionOrConstant rhs_batch_dimension_or_constant( in __anonbbc1c2d10202()