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 | 40 Tensor shape_constant(out_dtype_, TensorShape({input_shape.dims()})); in Compile() local
41 OP_REQUIRES_OK(ctx, TensorShapeToConstant(input_shape, &shape_constant)); in Compile()
42 ctx->SetConstantOutput(0, shape_constant); in Compile()
60 Tensor shape_constant(out_dtype_, TensorShape({input_shape.dims()})); in Compile() local
61 OP_REQUIRES_OK(ctx, TensorShapeToConstant(input_shape, &shape_constant)); in Compile()
62 ctx->SetConstantOutput(i, shape_constant); in Compile()
|
| D | shape_util.h | 30 Tensor* shape_constant);
|
| D | variable_ops.cc | 53 Tensor shape_constant(out_dtype_, TensorShape({shape.dims()})); in Compile() local
54 OP_REQUIRES_OK(ctx, TensorShapeToConstant(shape, &shape_constant)); in Compile()
55 ctx->SetConstantOutput(0, shape_constant); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 94 llvm::Constant* shape_constant = llvm::ConstantExpr::getBitCast( in HandleConstant() local
97 return IrArray(shape_constant, constant->shape()) in HandleConstant()
|