Searched refs:shape_values (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 57 llvm::SmallVector<Value, 4> shape_values; in CalculateShapeValue() local
59 shape_values.reserve(rank); in CalculateShapeValue()
61 shape_values.push_back(rewriter.create<mlir::DimOp>(loc, operand, i)); in CalculateShapeValue()
63 return rewriter.create<tensor::FromElementsOp>(loc, shape_values); in CalculateShapeValue()
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | resolve_svdf.cc | 85 std::vector<int32> shape_values( in InferFilterRank() local
88 reinterpret_cast<char*>(shape_values.data())); in InferFilterRank()
89 CHECK_EQ(shape_values.size(), 3); in InferFilterRank()
92 CHECK_EQ(shape_values[2], -1); in InferFilterRank()
93 return shape_values[1]; in InferFilterRank()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | broadcast_to_test.cc | 46 std::initializer_list<ShapeType> shape_values) { in BroadcastToOpModel() argument
49 AddConstInput(GetTensorType<ShapeType>(), shape_values, shape_shape); in BroadcastToOpModel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1064 def _total_size(shape_values): argument
1071 for val in shape_values:
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 3176 SmallVector<Value, 4> shape_values; in deriveShapeFromFirstOperand() local
3177 shape_values.reserve(operand_type.getRank()); in deriveShapeFromFirstOperand()
3182 shape_values.push_back( in deriveShapeFromFirstOperand()
3185 shape_values.push_back(builder->create<ConstantOp>( in deriveShapeFromFirstOperand()
3190 builder->create<tensor::FromElementsOp>(loc, shape_values)}; in deriveShapeFromFirstOperand()
|