Home
last modified time | relevance | path

Searched refs:tensor_store (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/tests/passes/
Dfusion_op_remover.mlir15 tensor_store %4, %out : memref<10xf32>
/external/tensorflow/tensorflow/core/kernels/
Dsession_ops.cc53 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); in Compute()
/external/llvm-project/mlir/integration_test/Dialect/Linalg/CPU/
Dtest-tensor-e2e.mlir16 // Instead of relying on tensor_store which introduces aliasing, we rely on
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.h104 TensorStore* tensor_store = nullptr; member
Ddirect_session.cc649 args.tensor_store = &run_state.tensor_store; in RunInternal()
765 if (!run_state.tensor_store.empty()) { in RunInternal()
766 TF_RETURN_IF_ERROR(run_state.tensor_store.SaveTensors( in RunInternal()
999 args.tensor_store = &run_state->tensor_store; in PRunSetup()
1085 s = run_state->tensor_store.SaveTensors(output_names, &session_state_); in PRun()
Ddirect_session.h204 TensorStore tensor_store; member
Dexecutor.cc402 tensor_store_(args.tensor_store), in ExecutorState()
703 params.tensor_store = tensor_store_; in Process()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc292 params.tensor_store = args.tensor_store; in Run()
/external/llvm-project/mlir/test/Dialect/Standard/
Dops.mlir100 tensor_store %1, %0 : memref<2xf32>
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h664 TensorStore* tensor_store = nullptr; member
1121 TensorStore* tensor_store() const { return params_->tensor_store; } in tensor_store() function
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dops.mlir707 // CHECK: tensor_store %[[VAR2]], %[[MEMREF:.*]] : memref<f32>
727 // CHECK: tensor_store %[[VAL0]], %{{.*}} : memref<f32>
728 // CHECK: tensor_store %[[VAL1]], %{{.*}} : memref<f32>
729 // CHECK: tensor_store %[[VAL2]], %{{.*}} : memref<f32>
/external/llvm-project/mlir/test/IR/
Dcore-ops.mlir888 // CHECK: tensor_store %[[TENSOR]], %[[MEMREF]] : memref<4x4xi32>
889 tensor_store %1, %0 : memref<4x4xi32>
897 // CHECK: tensor_store %[[TENSOR]], %[[MEMREF]] : memref<*xi32>
898 tensor_store %1, %0 : memref<*xi32>
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc288 params->tensor_store = nullptr; in PrepOpKernel()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_ops.mlir465 tensor_store %4, %out : memref<10xf32>
/external/llvm-project/mlir/docs/
DLangRef.md84 tensor_store %A to %A_m : memref<100x?xf32>
87 tensor_store %B to %B_m : memref<?x50xf32>
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/
DOps.td3559 def TensorStoreOp : Std_Op<"tensor_store",
3575 tensor_store %8, %10 : memref<4x?xf32, #layout, memspace0>