Searched refs:shape_constant (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_util.cc | 25 Tensor* shape_constant) { in TensorShapeToConstant() argument
27 if (shape_constant->dtype() == DT_INT32) { in TensorShapeToConstant()
28 auto vec = shape_constant->vec<int32>(); in TensorShapeToConstant()
39 auto vec = shape_constant->vec<int64>(); in TensorShapeToConstant()
|
| D | shape_op.cc | 59 Tensor shape_constant(out_dtype_, TensorShape({input_shape.dims()})); in Compile() local
60 OP_REQUIRES_OK(ctx, TensorShapeToConstant(input_shape, &shape_constant)); in Compile()
61 ctx->SetConstantOutput(0, shape_constant); in Compile()
170 Tensor shape_constant(out_dtype_, TensorShape({input_shape.dims()})); in Compile() local
172 TensorShapeToConstant(input_shape, &shape_constant)); in Compile()
173 ctx->SetConstantOutput(i, shape_constant); in Compile()
|
| D | shape_util.h | 30 Tensor* shape_constant);
|
| D | variable_ops.cc | 54 Tensor shape_constant(out_dtype_, TensorShape({shape.dims()})); in Compile() local
55 OP_REQUIRES_OK(ctx, TensorShapeToConstant(shape, &shape_constant)); in Compile()
56 ctx->SetConstantOutput(0, shape_constant); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 100 llvm::Constant* shape_constant = in HandleConstant() local
104 return IrArray(shape_constant, constant->shape()) in HandleConstant()
|